#!/usr/bin/env python
# coding: utf-8

import requests
from PIL import Image
from io import BytesIO

# 地理编码

def geocode(key,formatted_address,city=None,batch="false",sig=None,output="json",callback=None):
    """将详细的结构化地址转换为高德经纬度坐标"""
    geocode_url = "https://restapi.amap.com/v3/geocode/geo?parameters"
    params = {
        "key": key,
        "address": formatted_address,
        "city": city,
        "batch": batch,
        "sig": sig,
        "output": output,
        "callback": callback
    }
    r = requests.get(geocode_url, params=params)
    return r.json()

# 逆地理编码

def re_geocode(key,location,poitype=None,radius=1000,extensions="base",batch="false",roadlevel=1,sig=None,output="json",callback=None,homeorcorp=0):
    """将经纬度转换为详细结构化的地址，且返回附近周边的POI、AOI信息"""
    re_geocode_url = "https://restapi.amap.com/v3/geocode/regeo?parameters"
    params = {"key": key,
              "location": location,
              "poitype": poitype,
              "radius": radius,
              "extensions": extensions,
              "batch": batch,
              "roadlevel":roadlevel,
              "sig": sig,
              "output": output,
              "callback": callback,
              "homeorcorp": homeorcorp
             }
    r = requests.get(re_geocode_url, params=params)
    return r.json()

# 步行路径规划

def walking(key,origin,destination,sig=None,output="json",callback=None):
    walking_url = "https://restapi.amap.com/v3/direction/walking?parameters"
    params = {
        "key": key,
        "origin": origin,
        "destination": destination,
        "sig": sig,
        "output": output,
        "callback": callback
    }
    r = requests.get(walking_url, params=params)
    return r.json()
# 公交路径规划

def busing(key,origin,destination,city,cityd=None,extensions="base",strategy=0,nightflag=0,date=None,time=None,sig=None,output="json",callback=None):
    busing_url = "https://restapi.amap.com/v3/direction/transit/integrated?parameters"
    params = {
        "key": key,
        "origin": origin,
        "destination": destination,
        "city": city,
        "cityd": cityd,
        "extensions": extensions,
        "strategy": strategy,
        "nightflag": nightflag,
        "date": date,
        "time": time,
        "sig": sig,
        "output": output,
        "callback": callback
    }
    r = requests.get(busing_url, params=params)
    return r.json()

# 驾车路径规划

def driving(key,origin,destination,originid=None,destinationid=None,origintype=None,destinationtype=None,strategy=0,waypoints=None,avoidpolygons=None,avoidroad=None,province=None,number=None,cartype=0,ferry=0,roadaggregation="false",nosteps=0,sig=None,output="json",callback=None,extensions="base"):
    """规划以小客车、轿车通勤出行的方案，并且返回通勤方案的数据"""
    driving_url = "https://restapi.amap.com/v3/direction/driving?parameters"
    params = {
        "key": key,
        "origin": origin,
        "destination": destination,
        "originid": originid,
        "destinationid": destinationid,
        "origintype": origintype,
        "destinationtype": destinationtype,
        "strategy": strategy,
        "waypoints": waypoints,
        "avoidpolygons": avoidpolygons,
        "avoidroad": avoidroad,
        "province": province,
        "number": number,
        "cartype": cartype,
        "ferry": ferry,
        "roadaggregation": roadaggregation,
        "nosteps": nosteps,
        "sig": sig,
        "output": output,
        "callback": callback,
        "extensions": extensions
    }
    r = requests.get(driving_url, params=params)
    return r.json()


# 骑行路径规划

def bicycling(key,origin,destination):
    """规划骑行通勤方案，规划时不会考虑路况；考虑天桥、单行线、封路等情况"""
    bicycling_url = "https://restapi.amap.com/v4/direction/bicycling?parameters"
    params = {
        "key": key,
        "origin": origin,
        "destination": destination,
    }
    r = requests.get(bicycling_url, params=params)
    return r.json()

# 行政区域查询

def district(key,keywords,subdistric,page=1,offset=20,extensions="base",district_filter="adcode",callback=None,output="json"):
    """根据用户输入的搜索条件可以帮助用户快速的查找特定的行政区域信息"""
    district_url = "https://restapi.amap.com/v3/config/district?parameters"
    params = {
        "key": key,
        "keywords": keywords,
        "subdistric": subdistric,
        "page": page,
        "offset": offset,
        "extensions": extensions,
        "filter": district_filter,
        "callback": callback,
        "output": output
    }
    r = requests.get(district_url, params=params)
    return r.json()

# 关键字搜索

def poi_keywords(key,keywords,types=None,city=None,citylimit="false",children=0,offset=20,page=1,extensions="base",sig=None,output="json",callback=None):
    """通过用POI的关键字进行条件搜索"""
    poi_url = "https://restapi.amap.com/v3/place/text?parameters"
    params = {
        "key": key,
        "keywords": keywords,
        "types": types,
        "city": city,
        "citylimit": citylimit,
        "children": children,
        "offset": offset,
        "page": page,
        "extensions": extensions,
        "sig": sig,
        "output": output,
        "callback": callback
    }
    r = requests.get(poi_url, params)
    return r.json()


# 周边搜索

def poi_around(key,location,keywords=None,types="050000|070000|120000",city=None,radius=3000,sortrule="distance",offset=20,page=1,extensions="base",sig=None,output="json",callback=None):
    """在用户传入经纬度坐标点附近，在设定的范围内，按照关键字或POI类型搜索"""
    poi_around_url = "https://restapi.amap.com/v3/place/around?parameters"
    params = {
        "key": key,
        "location": location,
        "keywords": keywords,
        "types": types,
        "city": city,
        "radius": radius,
        "sortrule": sortrule,
        "offset": offset,
        "page": page,
        "extensions": extensions,
        "sig": sig,
        "output": output,
        "callback": callback
    }
    r = requests.get(poi_around_url, params)
    return r.json()


# 多边形搜索

def poi_polygon(key,polygon,keywords=None,types="120000|150000",offset=20,page=1,extensions="base",sig=None,output="json",callback=None):
    """在多边形区域内进行搜索POI"""
    poi_polygon_url = "https://restapi.amap.com/v3/place/polygon?parameters"
    params = {
        "key": key,
        "polygon": polygon,
        "keywords": keywords,
        "types": types,
        "offset": offset,
        "page": page,
        "extensions": extensions,
        "sig": sig,
        "output": output,
        "callback": callback
    }
    r = requests.get(poi_polygon_url, params)
    return r.json()

# IP定位

def ip(key,ip,sig=None,output="json"):
    """根据用户输入的IP地址，能够快速的帮用户定位IP的所在位置"""
    ip_url = "https://restapi.amap.com/v3/ip?parameters"
    params = {
        "key": key,
        "ip": ip,
        "sig": sig,
        "output": output}
    r = requests.get(ip_url, params=params)
    return r.json()

# 静态地图

def staticmap(key,location,zoom=15,size="400*400",scale=1,markers=None,labels=None,paths=None,traffic=0,sig=None):
    """静态地图服务通过返回一张地图图片响应HTTP请求，使用户能够将高德地图以图片形式嵌入自己的网页中。用户可以指定请求的地图位置、图片大小、以及在地图上添加覆盖物，如标签、标注、折线、多边形。"""
    from PIL import Image
    from io import BytesIO
    staticmap_url = "https://restapi.amap.com/v3/staticmap?parameters"
    params = {
        "key": key,
        "location": location,
        "zoom": zoom,
        "size": size,
        "scale": scale,
        "markers": markers,
        "labels": labels,
        "paths": paths,
        "traffic": traffic,
        "sig": sig
    }
    r = requests.get(staticmap_url, params)
    data = Image.open(BytesIO(r.content))
    return data


# 坐标转换

def convert(key,location,coordsys="autonavi",sig=None,output="json"):
    """将用户输入的非高德坐标（GPS坐标、mapbar坐标、baidu坐标）转换成高德坐标"""
    convert_url = "https://restapi.amap.com/v3/assistant/coordinate/convert?parameters"
    params = {
        "key": key,
        "locations": location,
        "coordsys": coordsys,
        "sig": sig,
        "output": output}
    r = requests.get(convert_url, params)
    return r.json()


# 天气查询

def weather(key,city,extension="all",output="json"):
    """根据用户输入的adcode，查询目标区域当前/未来的天气情况"""
    weather_url = "https://restapi.amap.com/v3/weather/weatherInfo?parameters"
    params = {
        "key": key,
        "city": city,
        "extensions":extension,
        "output": output
    }
    r = requests.get(weather_url, params)
    return r.json()


# 输入提示

def input_tips(key,keyword,poi_type=None,location=None,city=None,citylimit="false",datatype="all",sig=None,output="json",callback=None):
    """提供根据用户输入的关键词查询返回建议列表"""
    input_tips_url = "https://restapi.amap.com/v3/assistant/inputtips?parameters"
    params = {
        "key": key,
        "keywords": keyword,
        "type": poi_type,
        "location": location,
        "city": city,
        "citylimit": citylimit,
        "datatype": datatype,
        "sig": sig,
        "output": output,
        "callback": callback
    }
    r = requests.get(input_tips_url, params)
    return r.json()


# 地理围栏

def creat_geofence():
    creat_geofence_url = 'https://restapi.amap.com/v4/geofence/meta?key={}'
    params = {
        "name":"测试围栏",
        "points":"116.670040,23.468340;116.669649,23.466642;116.671216,23.466342;116.672627,23.468448;",
        "enable":"true",
        "repeat": "Mon,Tues,Wed,Thur,Fri,Sat,Sun",
#         "valid_time":"2020-4-23"
    }
    r = requests.post(creat_geofence_url.format(my_key),json=params)
    return r.json()

